import os
import pre_process
import shutil


if __name__ == "__main__":
    shutil.rmtree("dataset/train")
    shutil.rmtree("dataset/test")
    shutil.rmtree("dataset/valid")

    pre_process.mkdir("dataset/train")
    pre_process.mkdir("dataset/test")
    pre_process.mkdir("dataset/valid")
    Name = pre_process.file_name("org_dataset/Task1/Task1_Training")
    print(Name)
    length = len(Name)

    for fi in Name:
        order=int(fi[0:3])
        fi = "org_dataset/Task1/Task1_Training/" + str(fi)
        print(fi)
        pre_process.form(order, fi)

   # def labelform(mode):
    #    with open("sign/{}.txt".format(mode),"w"):
     #       a=



    # os.makedirs("test_"+str(i))
